package com.lihepeng.leecode.tree;

import org.junit.Test;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * 给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。
 * <p>
 * 例如：
 * 给定二叉树 [3,9,20,null,null,15,7],
 * <p>
 * 3
 * / \
 * 9  20
 * /  \
 * 15   7
 * 返回锯齿形层次遍历如下：
 * <p>
 * [
 * [3],
 * [20,9],
 * [15,7]
 * ]
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Solution103 {
    /**
     * 借助于栈结构完成锯齿遍历
     *
     * @param root
     * @return
     */
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) {
            return null;
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int level = 0;
        while (!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
            int currentSize = queue.size();
            for (int i = 1; i <= currentSize; i++) {
                TreeNode node = queue.poll();
                list.add(node.val);
                if (level % 2 == 0) {
                    // 偶数从左向右
                    if (node.right != null) {
                        queue.addLast(node.right);
                    }
                    if (node.left != null) {
                        queue.addLast(node.left);
                    }
                } else {
                    // 奇数从右向左
                    if (node.left != null) {
                        queue.addFirst(node.left);
                    }
                    if (node.right != null) {
                        queue.addFirst(node.right);
                    }

                }
            }
            level++;
            res.add(list);
        }
        return res;
    }

    /**
     * 每日一题
     * 2020年12月22日
     *
     * @param root
     * @return
     */
    public List<List<Integer>> zigzagLevelOrder01(TreeNode root) {
        if (root == null) {
            return null;
        }
        List<List<Integer>> resultList = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        // 判断当前行数是奇数还是偶数 奇数从左向右 偶数从有右向左
        int level = 0;
        while (!queue.isEmpty()) {

        }
        return null;
    }

    @Test
    public void runTest() {
        TreeNode head = new TreeNode(3);
        TreeNode h1 = new TreeNode(9);
        TreeNode h2 = new TreeNode(20);
        TreeNode h3 = new TreeNode(15);
        TreeNode h4 = new TreeNode(7);
        head.left = h1;
        head.right = h2;
        h2.left = h3;
        h2.right = h4;
        List<List<Integer>> lists = this.zigzagLevelOrder(head);
        System.out.println(lists);
    }
}
